package net.xinshi.utils;

import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang.StringUtils;

import java.io.UnsupportedEncodingException;

public class MD5SignUtils {
    private static final String DEFAULT_CHARSET = "UTF-8";

    /**
     * 签名
     *
     * @param srcStr
     * @RETURN
     */
    public static String sign(String srcStr) throws UnsupportedEncodingException {
        return sign(srcStr, DEFAULT_CHARSET);
    }

    /**
     * 签名
     *
     * @param srcStr
     * @param charset 字符编码
     * @return
     */
    public static String sign(String srcStr, String charset) throws UnsupportedEncodingException {
        return DigestUtils.md5Hex(getContentBytes(srcStr, charset));
    }

    /**
     * 签名字符串
     *
     * @param text          需要签名的字符串
     * @param sign          签名结果
     * @param input_charset 编码格式
     * @return 签名结果
     */
    public static boolean verify(String text, String sign, String input_charset) {
        String mysign = DigestUtils.md5Hex(getContentBytes(text, input_charset));
        if (StringUtils.equalsIgnoreCase(mysign,sign)) {
            return true;
        } else {
            return false;
        }
    }

    /**
     * @param content
     * @param charset
     * @return
     * @throws UnsupportedEncodingException
     */
    private static byte[] getContentBytes(String content, String charset) {
        if (charset == null || "".equals(charset)) {
            return content.getBytes();
        }
        try {
            return content.getBytes(charset);
        } catch (UnsupportedEncodingException e) {
            throw new RuntimeException("MD5签名过程中出现错误,指定的编码集不对,您目前指定的编码集是:" + charset);
        }
    }
}
